/**
 * Software License, Version 1.0
 * 
 * Copyright 2003 The Trustees of Indiana University.  All rights reserved.
 * 
 *
 *Redistribution and use in source and binary forms, with or without 
 *modification, are permitted provided that the following conditions are met:
 *
 *1) All redistributions of source code must retain the above copyright notice,
 * the list of authors in the original source code, this list of conditions and
 * the disclaimer listed in this license;
 *2) All redistributions in binary form must reproduce the above copyright 
 * notice, this list of conditions and the disclaimer listed in this license in
 * the documentation and/or other materials provided with the distribution;
 *3) Any documentation included with all redistributions must include the 
 * following acknowledgement:
 *
 *"This product includes software developed by the Community Grids Lab. For 
 * further information contact the Community Grids Lab at 
 * http://communitygrids.iu.edu/."
 *
 * Alternatively, this acknowledgement may appear in the software itself, and 
 * wherever such third-party acknowledgments normally appear.
 * 
 *4) The name Indiana University or Community Grids Lab or NaradaBrokering, 
 * shall not be used to endorse or promote products derived from this software 
 * without prior written permission from Indiana University.  For written 
 * permission, please contact the Advanced Research and Technology Institute 
 * ("ARTI") at 351 West 10th Street, Indianapolis, Indiana 46202.
 *5) Products derived from this software may not be called NaradaBrokering, 
 * nor may Indiana University or Community Grids Lab or NaradaBrokering appear
 * in their name, without prior written permission of ARTI.
 * 
 *
 * Indiana University provides no reassurances that the source code provided 
 * does not infringe the patent or any other intellectual property rights of 
 * any other entity.  Indiana University disclaims any liability to any 
 * recipient for claims brought by any other entity based on infringement of 
 * intellectual property rights or otherwise.  
 *
 *LICENSEE UNDERSTANDS THAT SOFTWARE IS PROVIDED "AS IS" FOR WHICH NO 
 *WARRANTIES AS TO CAPABILITIES OR ACCURACY ARE MADE. INDIANA UNIVERSITY GIVES
 *NO WARRANTIES AND MAKES NO REPRESENTATION THAT SOFTWARE IS FREE OF 
 *INFRINGEMENT OF THIRD PARTY PATENT, COPYRIGHT, OR OTHER PROPRIETARY RIGHTS. 
 *INDIANA UNIVERSITY MAKES NO WARRANTIES THAT SOFTWARE IS FREE FROM "BUGS", 
 *"VIRUSES", "TROJAN HORSES", "TRAP DOORS", "WORMS", OR OTHER HARMFUL CODE.  
 *LICENSEE ASSUMES THE ENTIRE RISK AS TO THE PERFORMANCE OF SOFTWARE AND/OR 
 *ASSOCIATED MATERIALS, AND TO THE PERFORMANCE AND VALIDITY OF INFORMATION 
 *GENERATED USING SOFTWARE.
 */
package cgl.narada.wsinfra.apps;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Calendar;
import java.util.Properties;
import java.util.StringTokenizer;

import javax.xml.soap.SOAPMessage;

import cgl.narada.wsinfra.deployment.Filter;
import cgl.narada.wsinfra.deployment.FilterPipeline;
import cgl.narada.wsinfra.deployment.FilterPipelineFactory;
import cgl.narada.wsinfra.deployment.NetworkSubstrate;
import cgl.narada.wsinfra.deployment.ServiceMessageListener;
import cgl.narada.wsinfra.deployment.impl.NBNetworkSubstrate;
import cgl.narada.wsinfra.exception.DeploymentException;
import cgl.narada.wsinfra.exception.MessageFlowException;
import cgl.narada.wsinfra.exception.ProcessingException;
import cgl.narada.wsinfra.schemas.addressing.EndpointReferenceType;
import cgl.narada.wsinfra.schemas.addressing.FromDocument;
import cgl.narada.wsinfra.schemas.soap.EnvelopeDocument;
import cgl.narada.wsinfra.util.SoapEnvelopeConversion;
import cgl.narada.wsinfra.util.SoapMessageAlteration;
import cgl.narada.wsinfra.util.SoapPrinter;
import cgl.narada.wsinfra.wsa.WsaEnvelopeCreator;
import cgl.narada.wsinfra.wsa.WsaEprCreator;
import cgl.narada.wsinfra.wsa.WsaProcessingFactory;
import cgl.narada.wsinfra.wse.impl.WseSinkProcessor;

public class WseSinkEndpoint implements ServiceMessageListener {
  private NetworkSubstrate networkSubstrate;
  private FilterPipelineFactory filterPipelineFactory;
  private FilterPipeline filterPipeline;
  private WsaEprCreator wsaEprCreator;
  private WsaEnvelopeCreator wsaEnvelopeCreator;
  private SoapEnvelopeConversion soapEnvelopeConversion;
  private SoapMessageAlteration soapMessageAlteration;
  private Filter loggingFilter;
  private Filter counterFilter;
  private Filter wseSinkFilter;
 
  private int tracker;
  private String sinkAddress;
 
  private WseSinkProcessor wseSinkProcessor;
  private String moduleName = "WseSinkEndpoint: ";

  public WseSinkEndpoint(String sinkAddress, Properties props) 
    throws DeploymentException {
    this.sinkAddress = sinkAddress;
    filterPipelineFactory =  FilterPipelineFactory.getInstance();
    filterPipeline = filterPipelineFactory.newFilterPipeline("WsePipeline");
    
    loggingFilter = new LoggingFilter();
    counterFilter = new CounterFilter();
    wseSinkFilter = new WseSinkFilter(sinkAddress);

    filterPipeline.addFilter(wseSinkFilter);
    filterPipeline.addFilter(counterFilter);
    filterPipeline.addFilter(loggingFilter);

    String pipelineInfo = moduleName + "Configured FilterPipeline" + 
      "\n\tWseSinkFilter=" + wseSinkFilter.getPositionInFilterPipeline() +
      "\n\tCounterFilter=" +  counterFilter.getPositionInFilterPipeline() +
      "\n\tLoggingFilter=" + loggingFilter.getPositionInFilterPipeline()+ "\n";

    System.out.println( pipelineInfo );
    
    networkSubstrate = new NBNetworkSubstrate();
    networkSubstrate.configureNetworking(props);
    
    networkSubstrate.registerPipeline(filterPipeline);
    filterPipeline.registerNetworkSubstrate(networkSubstrate);

    wsaEprCreator = WsaProcessingFactory.getWsaEprCreator();
    wsaEnvelopeCreator = WsaProcessingFactory.getWsaEnvelopeCreator();

    soapEnvelopeConversion = SoapEnvelopeConversion.getInstance();
    soapMessageAlteration = SoapMessageAlteration.getInstance();


    wseSinkProcessor = WseSinkProcessor.getInstance();
  }

  public void onMessage(SOAPMessage soapMessage) {
    System.out.println(moduleName + "Received SOAP Message from the " +
		       "underlying infrastructure");
    printSoapMessage(soapMessage);
  }

  public void onException(Exception exception) {


  }

  public void changeFilterOrder() throws DeploymentException {
    int position = loggingFilter.getPositionInFilterPipeline() ;
    if (position == 0) {
      filterPipeline.moveFilter(loggingFilter, 1);
    } 

    if (position == 1) {
      filterPipeline.moveFilter(loggingFilter, 0);
    } 
    
  }

  public void printSoapMessage(SOAPMessage soapMessage) {
    String soapMsg = null;
    try {
      soapMsg = SoapPrinter.getStringRepresentation(soapMessage);
    } catch (Exception e) {
      System.out.println(moduleName + "Problems printing SOAPMessage"); 
    }
    System.out.println(soapMsg);
  }

  public String createLinkTo(Properties props, String linkType) 
    throws DeploymentException { 
    if (networkSubstrate == null) {
      System.out.println(moduleName + "The network substrate is null");
    }
    String linkId = 
      ( (NBNetworkSubstrate)networkSubstrate).createLinkTo(props, linkType);
    return linkId;
  }


  public void sendSOAPMessage(SOAPMessage soapMessage) 
    throws DeploymentException, MessageFlowException, ProcessingException {
    filterPipeline.processMessageFromApplication(soapMessage); 
  }


  public void issueSubscribeRequest(String subscriptionType) 
    throws ProcessingException, MessageFlowException  {

    String sourceAddress = "tcp://deccan.ucs.indiana.edu:19000";
    EndpointReferenceType sourceEpr = wsaEprCreator.createEpr(sourceAddress);
    String deliveryMode = "http://schemas.xmlsoap.org/ws/2004/08/eventing/DeliveryModes/Push";
    String topicFilterDialect = "http://www.naradabrokering.org/TopicMatching";
    String regExFilterDialect = "http://www.naradabrokering.org/RegularExpressionMatching";

    String xpathFilterDialect = "http://www.w3.org/TR/1999/REC-xpath-19991116";
    String xqueryFilterDialect = "http://www.w3.org/TR/2004/WD-xquery-20041029";
    String topicFilterConstraint = "/Literature/Shakespere";
    String regExFilterConstraint = "[Tt]he [Qq]uick [Bb]rown [Ff]ox [Jj]umps";

    //String filterConstraint = "declare namespace s11='http://schemas.xmlsoap.org/soap/envelope/' c1='http://www.naradabrokering.org/catalog'" + "$this/s11:Envelope/s11:Body" + "/c1:catalog/c1:cd/c1:price";

    String xpathFilterConstraint = "declare namespace c1='http://www.naradabrokering.org/catalog'" + "$this//c1:catalog/c1:cd/c1:price > 20";

    

    String nsText = "declare namespace po = 'http://openuri.org/easypo'";
    String pathText = "$this//po:purchase-order/po:line-item";
      //[po:price <= 20.00]";
    String queryText = nsText + pathText;
    String xqueryFilterConstraint = queryText;

    String filterConstraint = topicFilterConstraint;
    String filterDialect = topicFilterDialect;

    if (subscriptionType.equals("topic")) {
      filterConstraint = topicFilterConstraint;
      filterDialect = topicFilterDialect;
    }

    if (subscriptionType.equals("regex")) {
      filterConstraint = regExFilterConstraint;
      filterDialect = regExFilterDialect;
    }

    if (subscriptionType.equals("xpath")) {
      filterConstraint = xpathFilterConstraint;
      filterDialect = xpathFilterDialect;
    }

    if (subscriptionType.equals("xquery")) {
      filterConstraint = xqueryFilterConstraint;
      filterDialect = xqueryFilterDialect;
    }

    EndpointReferenceType sinkEpr = wsaEprCreator.createEpr(sinkAddress);
    EndpointReferenceType endTo = wsaEprCreator.createEpr(sinkAddress);
    Calendar expiresAt = Calendar.getInstance();
    expiresAt.add(Calendar.MONTH, 1);

    EnvelopeDocument envelopeDocument = 
      wseSinkProcessor.createSubscribeRequest(sourceEpr, deliveryMode, endTo,
					      filterDialect, filterConstraint,
					      expiresAt);
    String report = moduleName + "The CREATED Subscription Request is\n" +
      envelopeDocument.toString();
    //System.out.println(report);
    SOAPMessage soapMessage = getSOAPMessage(envelopeDocument);
    wseSinkFilter.enrouteToNetwork(soapMessage);
  }
  

  public void issueGetStatusRequest(String subscriptionIdentifier) 
    throws ProcessingException, MessageFlowException {
    EnvelopeDocument envelopeDocument = 
      wseSinkProcessor.createGetStatus(subscriptionIdentifier);
    String report = moduleName + "The CREATED GetStatus Request is\n" +
      envelopeDocument.toString();
    System.out.println(report);
    SOAPMessage soapMessage = getSOAPMessage(envelopeDocument);
    wseSinkFilter.enrouteToNetwork(soapMessage);
  }


  public void issueRenewRequest(String subscriptionIdentifier) 
    throws ProcessingException, MessageFlowException {
    Calendar expiresAt = Calendar.getInstance();
    expiresAt.add(Calendar.MONTH, 2);
    EnvelopeDocument envelopeDocument = 
      wseSinkProcessor.createRenew(subscriptionIdentifier, expiresAt);
    String report = moduleName + "The CREATED Renew Request is\n" +
      envelopeDocument.toString();
    System.out.println(report);
    SOAPMessage soapMessage = getSOAPMessage(envelopeDocument);
    wseSinkFilter.enrouteToNetwork(soapMessage);
  }


  public void issueUnsubscribeRequest(String subscriptionIdentifier) 
    throws ProcessingException, MessageFlowException {
   
    EnvelopeDocument envelopeDocument = 
      wseSinkProcessor.createUnsubscribe(subscriptionIdentifier);
    String report = moduleName + "The CREATED Unsubscribe Request is\n" +
      envelopeDocument.toString();
    System.out.println(report);
    SOAPMessage soapMessage = getSOAPMessage(envelopeDocument);
    wseSinkFilter.enrouteToNetwork(soapMessage);
  }


  private SOAPMessage getSOAPMessage(EnvelopeDocument envelopeDocument) {
    SOAPMessage soapMessage = null;
    try {
      soapMessage =  soapEnvelopeConversion.getSOAPMessage(envelopeDocument);
    } catch (ProcessingException ex) {
      String errorReport ="Problems converting to javax.xml.SOAPMessage\n"
	+ ex.toString();
      System.out.println(errorReport);
    }
    
    return soapMessage;
  }



  public SOAPMessage createSOAPMessage(String source) {
    if (source == null) {
      System.out.println(moduleName + "Specified SourceAddress is NULL!");
      return null;
    }
    EndpointReferenceType sourceEpr = wsaEprCreator.createEpr(source);
    FromDocument from = FromDocument.Factory.newInstance();
    from.setFrom(sourceEpr);
    EnvelopeDocument envelopeDocument = 
      wsaEnvelopeCreator.createSoapEnvelope(from);
    
    return getSOAPMessage(envelopeDocument);
  }


  public static void main(String[] args) {
    if (args.length != 2) {
      String errorReport = 
	"Usage: java cgl.narada.wsinfra.apps.WseSinkEndpoint " +
	"[hostname] [TCPServerPort] ";
      System.out.println(errorReport);
      System.exit(0);
    }

    String hostname = args[0];
    String port = args[1];
    Properties props = new Properties();
    props.put("TCPServerPort", port);
    
    BufferedReader r = new BufferedReader(new InputStreamReader(System.in));
    StringTokenizer tok;
    String cmd;

    try {
      String sinkAddress = "tcp://" + hostname + ":" + port;
      WseSinkEndpoint endpoint = 
	new WseSinkEndpoint(sinkAddress, props);
      System.out.println("Listening to TCP communications on port [" + port 
			 + "]. \n Type h for help on commands");
      while (true) {
      	cmd = r.readLine();
	
	if (cmd == null) {
	  System.out.println("Terminating program ....");
	  break;
	}

	if(cmd.startsWith("h")) {
	  System.out.println("Create Link \t\t-> cr hostname portNum ");
	  System.out.println("Subscribe   \t\t-> sub <subType> [t][r][x][q]" );
	  System.out.println("GetStatus   \t\t-> status [subscriptionId]" );
	  System.out.println("Renew       \t\t-> renew [subscriptionId]" );
	  System.out.println("UnSubscribe \t\t-> unsub [subscriptionId]" );
	  System.out.println("Command-list\t\t-> h]" );
	}
	
	if (cmd.startsWith("cr")) {
	  tok = new StringTokenizer(cmd, " ");
	  tok.nextToken(); /* skipping the token "cr" */
	  Properties linkProps = new Properties();
	  linkProps.put("hostname", tok.nextToken());
	  linkProps.put("portnum", tok.nextToken());
	  String _linkId = endpoint.createLinkTo(linkProps, "tcp");
	  System.out.println("Created link with id => " + _linkId); 
	} /** end create link */
	
	if (cmd.startsWith("sub")) {
	  tok = new StringTokenizer(cmd, " ");
	  tok.nextToken(); /* skipping the token "status" */
	  String subscriptionType = "topic"; 
	  String widget = tok.nextToken();
	  if (widget.startsWith("t")) {
	    subscriptionType = "topic"; 
	  }
	  if (widget.startsWith("x")) {
	    subscriptionType = "xpath"; 
	  }
	  if (widget.startsWith("r")) {
	    subscriptionType = "regex"; 
	  }
	  if (widget.startsWith("q")) {
	    subscriptionType = "xquery"; 
	  }	  

	  endpoint.issueSubscribeRequest(subscriptionType);
	}

	if (cmd.startsWith("status")) {
	  tok = new StringTokenizer(cmd, " ");
	  tok.nextToken(); /* skipping the token "status" */
	  String subscriptionIdentifier = tok.nextToken();
	  endpoint.issueGetStatusRequest(subscriptionIdentifier);
	}

	if (cmd.startsWith("renew")) {
	  tok = new StringTokenizer(cmd, " ");
	  tok.nextToken(); /* skipping the token "status" */
	  String subscriptionIdentifier = tok.nextToken();
	  endpoint.issueRenewRequest(subscriptionIdentifier);
	}
	
	if (cmd.startsWith("unsub")) {
	  tok = new StringTokenizer(cmd, " ");
	  tok.nextToken(); /* skipping the token "status" */
	  String subscriptionIdentifier = tok.nextToken();
	  endpoint.issueUnsubscribeRequest(subscriptionIdentifier);
	}
	
      } /* end while(true) */

    }catch (Exception ioe) {
      ioe.printStackTrace();
    } /* end try-catch */
  }/*end main() */

}
